package com.guigu.mm;

import java.io.IOException;
import java.io.OutputStream;
import java.util.List;

import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFFont;
import org.apache.poi.hssf.usermodel.HSSFRichTextString;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.BorderStyle;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.ss.usermodel.HorizontalAlignment;
import org.apache.poi.ss.usermodel.VerticalAlignment;
import org.apache.poi.ss.util.CellRangeAddress;

//import javax.servlet.http.HttpServletResponse;

/*******************************************
 *
 * @Package com.cccuu.project.myUtils
 * @Author duan
 * @Date 2018/3/29 9:12
 * @Version V1.0
 *******************************************/
public class ExcelUtil {

	/**
	 * 导出excel
	 * 
	 * @param title    导出表的标题
	 * @param rowsName 导出表的列名
	 * @param dataList 需要导出的数据
	 * @param fileName 生成excel文件的文件名
	 * @param response
	 */
	public void exportExcel(String title, String[] rowsName, List<Object[]> dataList, String fileName)
			throws Exception {
//		OutputStream output = response.getOutputStream();
//		HttpServletResponse response
//		response.reset();
//		response.setHeader("Content-disposition", "attachment; filename=" + fileName);
//		response.setContentType("application/msexcel");
//		this.export(title, rowsName, dataList, fileName, output);
//		this.close(output);

	}

	/*
	 * 导出数据
	 */
	private void export(String title, String[] rowName, List<Object[]> dataList, String fileName, OutputStream out)
			throws Exception {
		try {
			HSSFWorkbook workbook = new HSSFWorkbook(); // 创建工作簿对象
			HSSFSheet sheet = workbook.createSheet(title); // 创建工作表
			HSSFRow rowm = sheet.createRow(0); // 产生表格标题行
			HSSFCell cellTiltle = rowm.createCell(0); // 创建表格标题列
			// sheet样式定义; getColumnTopStyle(); getStyle()均为自定义方法 --在下面,可扩展
			HSSFCellStyle columnTopStyle = this.getColumnTopStyle(workbook);// 获取列头样式对象
			HSSFCellStyle style = this.getStyle(workbook); // 获取单元格样式对象
			// 合并表格标题行，合并列数为列名的长度,第一个0为起始行号，第二个1为终止行号，第三个0为起始列好，第四个参数为终止列号
			sheet.addMergedRegion(new CellRangeAddress(0, 1, 0, (rowName.length - 1)));
			cellTiltle.setCellStyle(columnTopStyle); // 设置标题行样式
			cellTiltle.setCellValue(title); // 设置标题行值
			int columnNum = rowName.length; // 定义所需列数
			HSSFRow rowRowName = sheet.createRow(2); // 在索引2的位置创建行(最顶端的行开始的第二行)
			// 将列头设置到sheet的单元格中
			for (int n = 0; n < columnNum; n++) {
				HSSFCell cellRowName = rowRowName.createCell(n); // 创建列头对应个数的单元格
//				cellRowName.setCellType(HSSFCell.CELL_TYPE_STRING); // 设置列头单元格的数据类型
				cellRowName.setCellType(CellType.STRING); // 设置列头单元格的数据类型
				HSSFRichTextString text = new HSSFRichTextString(rowName[n]);
				cellRowName.setCellValue(text); // 设置列头单元格的值
				cellRowName.setCellStyle(columnTopStyle); // 设置列头单元格样式
			}

			// 将查询出的数据设置到sheet对应的单元格中
			for (int i = 0; i < dataList.size(); i++) {
				Object[] obj = dataList.get(i); // 遍历每个对象
				HSSFRow row = sheet.createRow(i + 3); // 创建所需的行数
				for (int j = 0; j < obj.length; j++) {
					HSSFCell cell = null; // 设置单元格的数据类型
					if (j == 0) {
//						cell = row.createCell(j, HSSFCell.CELL_TYPE_NUMERIC);
						cell = row.createCell(j, CellType.NUMERIC);
						cell.setCellValue(i + 1);
					} else {
//						cell = row.createCell(j, HSSFCell.CELL_TYPE_STRING);
						cell = row.createCell(j, CellType.STRING);
						if (!"".equals(obj[j]) && obj[j] != null) {
							cell.setCellValue(obj[j].toString()); // 设置单元格的值
						}
					}
					cell.setCellStyle(style); // 设置单元格样式
				}
			}

			// 让列宽随着导出的列长自动适应
			for (int colNum = 0; colNum < columnNum; colNum++) {
				int columnWidth = sheet.getColumnWidth(colNum) / 256;
				for (int rowNum = 0; rowNum < sheet.getLastRowNum(); rowNum++) {
					HSSFRow currentRow;
					// 当前行未被使用过
					if (sheet.getRow(rowNum) == null) {
						currentRow = sheet.createRow(rowNum);
					} else {
						currentRow = sheet.getRow(rowNum);
					}
					if (currentRow.getCell(colNum) != null) {
						HSSFCell currentCell = currentRow.getCell(colNum);
//						if (currentCell.getCellType() == HSSFCell.CELL_TYPE_STRING) {
						if (currentCell.getCellType() == CellType.STRING) {
							int length = currentCell.getStringCellValue().getBytes().length;
							if (columnWidth < length) {
								columnWidth = length;
							}
						}
					}
				}
				if (colNum == 0) {
					sheet.setColumnWidth(colNum, (columnWidth - 2) * 256);
				} else {
					sheet.setColumnWidth(colNum, (columnWidth + 4) * 256);
				}
			}
			workbook.write(out);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	/*
	 * 列头单元格样式
	 */
	private HSSFCellStyle getColumnTopStyle(HSSFWorkbook workbook) {

		// 设置字体
		HSSFFont font = workbook.createFont();
		// 设置字体大小
		font.setFontHeightInPoints((short) 11);
		// 字体加粗
//		font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
		font.setBold(true);
		// 设置字体名字
		font.setFontName("Courier New");
		// 设置样式;
		HSSFCellStyle style = workbook.createCellStyle();
		// 设置底边框;
//		style.setBorderBottom(HSSFCellStyle.BORDER_THIN);
		style.setBorderBottom(BorderStyle.THIN);
		// 设置底边框颜色;
//		style.setBottomBorderColor(HSSFColor.BLACK.index);
		style.setBottomBorderColor((short) 2);
		// 设置左边框;
//		style.setBorderLeft(HSSFCellStyle.BORDER_THIN);
		style.setBorderLeft(BorderStyle.THIN);
		// 设置左边框颜色;
//		style.setLeftBorderColor(HSSFColor.BLACK.index);
		style.setLeftBorderColor((short) 2);
		// 设置右边框;
//		style.setBorderRight(HSSFCellStyle.BORDER_THIN);
		style.setBorderRight(BorderStyle.THIN);
		// 设置右边框颜色;
//		style.setRightBorderColor(HSSFColor.BLACK.index);
		style.setRightBorderColor((short) 2);
		// 设置顶边框;
//		style.setBorderTop(HSSFCellStyle.BORDER_THIN);
		style.setBorderTop(BorderStyle.THIN);
		// 设置顶边框颜色;
//		style.setTopBorderColor(HSSFColor.BLACK.index);
		style.setTopBorderColor((short) 2);
		// 在样式用应用设置的字体;
		style.setFont(font);
		// 设置自动换行;
		style.setWrapText(false);
		// 设置水平对齐的样式为居中对齐;
//		style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
		style.setAlignment(HorizontalAlignment.CENTER);
		// 设置垂直对齐的样式为居中对齐;
//		style.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
		style.setVerticalAlignment(VerticalAlignment.CENTER);

		return style;

	}

	/*
	 * 列数据信息单元格样式
	 */
	private HSSFCellStyle getStyle(HSSFWorkbook workbook) {
		// 设置字体
		HSSFFont font = workbook.createFont();
		// 设置字体大小
		// font.setFontHeightInPoints((short)10);
		// 字体加粗
		// font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
		// 设置字体名字
		font.setFontName("Courier New");
		// 设置样式;
		HSSFCellStyle style = workbook.createCellStyle();
		// 设置底边框;
//		style.setBorderBottom(HSSFCellStyle.BORDER_THIN);
		style.setBorderBottom(BorderStyle.THIN);
		// 设置底边框颜色;
//		style.setBottomBorderColor(HSSFColor.BLACK.index);
		style.setBottomBorderColor((short) 2);
		// 设置左边框;
//		style.setBorderLeft(HSSFCellStyle.BORDER_THIN);
		style.setBorderLeft(BorderStyle.THIN);
		// 设置左边框颜色;
//		style.setLeftBorderColor(HSSFColor.BLACK.index);
		style.setLeftBorderColor((short) 2);
		// 设置右边框;
//		style.setBorderRight(HSSFCellStyle.BORDER_THIN);
		style.setBorderRight(BorderStyle.THIN);
		// 设置右边框颜色;
//		style.setRightBorderColor(HSSFColor.BLACK.index);
		style.setRightBorderColor((short) 2);
		// 设置顶边框;
//		style.setBorderTop(HSSFCellStyle.BORDER_THIN);
		style.setBorderTop(BorderStyle.THIN);
		// 设置顶边框颜色;
//		style.setTopBorderColor(HSSFColor.BLACK.index);
		style.setTopBorderColor((short) 2);
		// 在样式用应用设置的字体;
		style.setFont(font);
		// 设置自动换行;
		style.setWrapText(false);
		// 设置水平对齐的样式为居中对齐;
//		style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
		style.setAlignment(HorizontalAlignment.CENTER);
		// 设置垂直对齐的样式为居中对齐;
//		style.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
		style.setVerticalAlignment(VerticalAlignment.CENTER);
		return style;
	}

	/**
	 * 关闭输出流
	 * 
	 * @param os
	 */
	private void close(OutputStream os) {
		if (os != null) {
			try {
				os.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}

}